package leetcode.hot100;

import leetcode.common.TreeNode;

import java.util.Deque;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

@SuppressWarnings("all")
public class _103_二叉树的锯齿形层序遍历_双端队列 {

    /**
     * 给定一个二叉树，返回其节点值的锯齿形层序遍历
     * （即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）
     * 3
     * / \
     * 9  20
     * /  \
     * 15   7
     * 返回锯齿形层序遍历如下：
     * [
     * [3],
     * [20,9],
     * [15,7]
     * ]
     */
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> result = new LinkedList<>();
        if (root == null) {
            return result;
        }
        Queue<TreeNode> nodeQueue = new LinkedList<>();
        nodeQueue.offer(root);
        boolean isOrderLeft = true;

        while (!nodeQueue.isEmpty()) {
            // 使用双端队列存储每一层遍历的结点
            Deque<Integer> levelList = new LinkedList<>();
            // 当前队列中的结点数量
            int size = nodeQueue.size();
            for (int i = 0; i < size; ++i) {
                TreeNode curNode = nodeQueue.poll();
                if (isOrderLeft) {
                    // 从左往右遍历 依次从尾部入队，可以确保第一个入队的结点在队首
                    levelList.offerLast(curNode.val);
                } else {
                    // 从右往左遍历 依次从头部入队，可以确保第一个入队的结点在队尾
                    levelList.offerFirst(curNode.val);
                }
                // 当前结点左右孩子入队
                if (curNode.left != null) {
                    nodeQueue.offer(curNode.left);
                }
                if (curNode.right != null) {
                    nodeQueue.offer(curNode.right);
                }
            }
            // 收集局部结果
            result.add(new LinkedList<>(levelList));
            // 左右交替
            isOrderLeft = !isOrderLeft;
        }

        return result;
    }
}
